Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_lxmf_sync.py

Displaying Raw • Download

Tff7b72import T7ee787json

Tff7b72import T7ee787shutil

Tff7b72import T7ee787tempfile

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72import T7ee787LXMF

Tff7b72import T7ee787pytest

Tff7b72import T7ee787RNS

Tff7b72from T7ee787meshchatxT7ee787.T7ee787meshchat Tff7b72import Te6edf3ReticulumMeshChat

T8b949e# Store original constants

Te6edf3PR_IDLE Tff7b72= Te6edf3LXMFTff7b72.Td2a8ffLXMRouterTff7b72.Td2a8ffPR_IDLE

Te6edf3PR_COMPLETE Tff7b72= Te6edf3LXMFTff7b72.Td2a8ffLXMRouterTff7b72.Td2a8ffPR_COMPLETE

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8fftemp_dirTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3dir_path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72yield Te6edf3dir_path

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Te6edf3dir_pathTb4b4b4)

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8ffmock_appTb4b4b4(Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Tff7b72with Tb4b4b4(

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.ReticulumTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_rnsTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.TransportTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffLXMF.LXMRouterTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_router_classTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.get_file_pathTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Ta5d6ff"Ta5d6ff/tmp/mock_pathTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.generate_ssl_certificateTa5d6ff"Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

T8b949e# Set up constants on the mock class

Te6edf3mock_router_classTff7b72.Td2a8ffPR_IDLE Tff7b72= Te6edf3PR_IDLE

Te6edf3mock_router_classTff7b72.Td2a8ffPR_COMPLETE Tff7b72= Te6edf3PR_COMPLETE

Te6edf3mock_router Tff7b72= Te6edf3mock_router_classTff7b72.Td2a8ffreturn_value

Te6edf3mock_routerTff7b72.Td2a8ffPR_IDLE Tff7b72= Te6edf3PR_IDLE

Te6edf3mock_routerTff7b72.Td2a8ffPR_COMPLETE Tff7b72= Te6edf3PR_COMPLETE

Te6edf3mock_routerTff7b72.Td2a8ffpropagation_transfer_state Tff7b72= Te6edf3PR_IDLE

Te6edf3mock_routerTff7b72.Td2a8ffpropagation_transfer_progress Tff7b72= T79c0ff0

Te6edf3mock_routerTff7b72.Td2a8ffpropagation_transfer_last_result Tff7b72= T79c0ff0

Te6edf3mock_dest Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_destTff7b72.Td2a8ffhexhash Tff7b72= Ta5d6ff"Ta5d6ffmock_hashTa5d6ff"

Te6edf3mock_routerTff7b72.Td2a8ffregister_delivery_identityTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_dest

Te6edf3mock_rns_inst Tff7b72= Te6edf3mock_rnsTff7b72.Td2a8ffreturn_value

Te6edf3mock_rns_instTff7b72.Td2a8fftransport_enabledTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72False

Tff7b72with Te6edf3patchTb4b4b4(

Ta5d6ff"Ta5d6ffmeshchatx.src.backend.meshchat_utils.LXMRouterTa5d6ff"Tb4b4b4,

Tb4b4b4) Tff7b72as Te6edf3mock_utils_routerTb4b4b4:

Te6edf3mock_utils_routerTff7b72.Td2a8ffPR_IDLE Tff7b72= Te6edf3PR_IDLE

Te6edf3mock_utils_routerTff7b72.Td2a8ffPR_COMPLETE Tff7b72= Te6edf3PR_COMPLETE

Te6edf3real_id Tff7b72= Te6edf3RNSTff7b72.Td2a8ffIdentityTb4b4b4(Tb4b4b4)

Te6edf3app Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(

Te6edf3identityTff7b72=Te6edf3real_idTb4b4b4,

Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,

Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,

Tb4b4b4)

Te6edf3appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffmessage_router Tff7b72= Te6edf3mock_router

Tff7b72with Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(

Te6edf3appTb4b4b4,

Ta5d6ff"Ta5d6ffsend_config_to_websocket_clientsTa5d6ff"Tb4b4b4,

Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4,

Tb4b4b4)Tb4b4b4:

Tff7b72yield Te6edf3app

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_lxmf_sync_endpointsTb4b4b4(Te6edf3mock_appTb4b4b4)Tb4b4b4:

T8b949e# 1. Test status endpoint initially idle

Te6edf3handler Tff7b72= Tff7b72None

Tff7b72for Te6edf3route Tff7b72in Te6edf3mock_appTff7b72.Td2a8ffget_routesTb4b4b4(Tb4b4b4)Tb4b4b4:

Tff7b72if Tb4b4b4(

Te6edf3routeTff7b72.Td2a8ffpath Tff7b72== Ta5d6ff"Ta5d6ff/api/v1/lxmf/propagation-node/statusTa5d6ff"

Tff7b72and Te6edf3routeTff7b72.Td2a8ffmethod Tff7b72== Ta5d6ff"Ta5d6ffGETTa5d6ff"

Tb4b4b4)Tb4b4b4:

Te6edf3handler Tff7b72= Te6edf3routeTff7b72.Td2a8ffhandler

Tff7b72break

Tff7b72assert Te6edf3handler Tff7b72is Tff7b72not Tff7b72None

Te6edf3response Tff7b72= Tff7b72await Te6edf3handlerTb4b4b4(Tff7b72NoneTb4b4b4)

Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3responseTff7b72.Td2a8ffbodyTb4b4b4)

Tff7b72assert Te6edf3dataTb4b4b4[Ta5d6ff"Ta5d6ffpropagation_node_statusTa5d6ff"Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffstateTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffidleTa5d6ff"

T8b949e# 2. Test sync initiation

Te6edf3sync_handler Tff7b72= Tff7b72None

Tff7b72for Te6edf3route Tff7b72in Te6edf3mock_appTff7b72.Td2a8ffget_routesTb4b4b4(Tb4b4b4)Tb4b4b4:

Tff7b72if Te6edf3routeTff7b72.Td2a8ffpath Tff7b72== Ta5d6ff"Ta5d6ff/api/v1/lxmf/propagation-node/syncTa5d6ff" Tff7b72and Te6edf3routeTff7b72.Td2a8ffmethod Tff7b72== Ta5d6ff"Ta5d6ffGETTa5d6ff"Tb4b4b4:

Te6edf3sync_handler Tff7b72= Te6edf3routeTff7b72.Td2a8ffhandler

Tff7b72break

Tff7b72assert Te6edf3sync_handler Tff7b72is Tff7b72not Tff7b72None

T8b949e# Mock outbound propagation node configured

Te6edf3mock_appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffmessage_routerTff7b72.Td2a8ffget_outbound_propagation_nodeTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffsomehashTa5d6ff"

Te6edf3response Tff7b72= Tff7b72await Te6edf3sync_handlerTb4b4b4(Tff7b72NoneTb4b4b4)

Tff7b72assert Te6edf3responseTff7b72.Td2a8ffstatus Tff7b72== T79c0ff200

Te6edf3mock_appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffmessage_routerTff7b72.Td2a8ffrequest_messages_from_propagation_nodeTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)

T8b949e# 3. Test status change to complete

Te6edf3mock_appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffmessage_routerTff7b72.Td2a8ffpropagation_transfer_state Tff7b72= Tb4b4b4(

Te6edf3LXMFTff7b72.Td2a8ffLXMRouterTff7b72.Td2a8ffPR_COMPLETE

Tb4b4b4)

Te6edf3response Tff7b72= Tff7b72await Te6edf3handlerTb4b4b4(Tff7b72NoneTb4b4b4)

Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3responseTff7b72.Td2a8ffbodyTb4b4b4)

Tff7b72assert Te6edf3dataTb4b4b4[Ta5d6ff"Ta5d6ffpropagation_node_statusTa5d6ff"Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffstateTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffcompleteTa5d6ff"

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_specific_node_hash_validationTb4b4b4(Te6edf3mock_appTb4b4b4)Tb4b4b4:

Te6edf3node_hash_hex Tff7b72= Ta5d6ff"Ta5d6ffd81255ae2ff367d4883b16c9cc8c6178Ta5d6ff"

T8b949e# Ensure update_config doesn't crash due to mock serialization

Tff7b72with Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3mock_appTb4b4b4, Ta5d6ff"Ta5d6ffsend_config_to_websocket_clientsTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4)Tb4b4b4:

T8b949e# Set the preferred propagation node

Tff7b72await Te6edf3mock_appTff7b72.Td2a8ffupdate_configTb4b4b4(

Tb4b4b4{Ta5d6ff"Ta5d6fflxmf_preferred_propagation_node_destination_hashTa5d6ff"Tb4b4b4: Te6edf3node_hash_hexTb4b4b4}Tb4b4b4,

Tb4b4b4)

T8b949e# Verify it was set on the router correctly as 16 bytes

Te6edf3expected_bytes Tff7b72= Tffa657bytesTff7b72.Td2a8fffromhexTb4b4b4(Te6edf3node_hash_hexTb4b4b4)

Te6edf3mock_appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffmessage_routerTff7b72.Td2a8ffset_outbound_propagation_nodeTff7b72.Td2a8ffassert_called_withTb4b4b4(

Te6edf3expected_bytesTb4b4b4,

Tb4b4b4)

T8b949e# Trigger sync

Te6edf3sync_handler Tff7b72= Tff7b72None

Tff7b72for Te6edf3route Tff7b72in Te6edf3mock_appTff7b72.Td2a8ffget_routesTb4b4b4(Tb4b4b4)Tb4b4b4:

Tff7b72if Te6edf3routeTff7b72.Td2a8ffpath Tff7b72== Ta5d6ff"Ta5d6ff/api/v1/lxmf/propagation-node/syncTa5d6ff" Tff7b72and Te6edf3routeTff7b72.Td2a8ffmethod Tff7b72== Ta5d6ff"Ta5d6ffGETTa5d6ff"Tb4b4b4:

Te6edf3sync_handler Tff7b72= Te6edf3routeTff7b72.Td2a8ffhandler

Tff7b72break

T8b949e# Ensure it's considered configured

Te6edf3mock_appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffmessage_routerTff7b72.Td2a8ffget_outbound_propagation_nodeTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3expected_bytes

Tff7b72await Te6edf3sync_handlerTb4b4b4(Tff7b72NoneTb4b4b4)

Te6edf3mock_appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffmessage_routerTff7b72.Td2a8ffrequest_messages_from_propagation_nodeTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)